from ninja import ModelSchema

from apps.commom.good import GoodsModelSchema, GoodSpecificationModelSchema
from models.models import VipOrderInfo, OrderInfo, AgentOrderInfo


class OrderInfoResponseSchema(ModelSchema):
    good: GoodsModelSchema = None
    good_spec: GoodSpecificationModelSchema = None

    class Config:
        model = OrderInfo
        model_fields = "__all__"


class VipOrderInfoResponseSchema(ModelSchema):
    class Config:
        model = VipOrderInfo
        model_fields = "__all__"


class AgentOrderInfoResponseSchema(ModelSchema):
    class Config:
        model = AgentOrderInfo
        model_fields = "__all__"
